/**
 * Created by Administrator on 2025/5/7.
 * */
#ifndef AIMTD_ERA
#define AIMTD_ERA

#include "CudaModule.h"
#include "../GlobalParameters.h"

extern "C" BaseObject *createAIMTD();
extern "C" void destroyAIMTD(BaseObject *p);

class AIMTD : public CudaModule {
public:
AIMTD();

~AIMTD();

//void onPushCudaOperations(buffer_table_t *input, buffer_table_t *output, cudaStream_t stream) override;
    void onIssueStreamedCudaOperations(buffer_table_t *input, buffer_table_t *output, cudaStream_t stream) override;
};

#endif //TASK2_ERA
